#!user/bin/env python3
# -- coding: utf-8 --
# @Createtime：2022/5/15
# @Updatetime：
# @Author：Test008
# @File：InsertStuInfo.py
# @Description：添加学生及成绩信息，将学生信息保存到文件中
from config import filename


def insert():
    student_list = []
    while True:
        id = input('请输入学号：如1001：')
        if not id:
            break
        name = input('请输入姓名：')
        if not name:
            break
        try:
            english = int(input('请输入英语成绩：'))
            python = int(input('请输入python成绩：'))
            java = int(input('请输入java成绩：'))
        except:
            print('输入无效，请输入整数！！！')
            continue
        # 输入内容添加到字典中
        student_dict = {'id': id, 'name': name, 'english': english, 'python': python, 'java': java}
        # 添加到列表
        student_list.append(student_dict)
        answer = input('是否需要继续添加？y/n\n')
        if answer == 'y':
            continue
        else:
            break

    save(student_list)
    print('输入的内容已保存到系统中！！！')


def save(list1):
    try:
        file = open(filename, 'a', encoding='utf-8')
    except:
        file = open(filename, 'w', encoding='utf-8')
    for i in list1:
        file.write(str(i) + '\n')
    file.close()
